#include "friendmodel.hpp"
#include "db.h"


void FriendModel::insert(int userid,int friendid){
    char sql[1024] = {0};

    sprintf(sql,"insert into chat_friend(userId,friendId) values(%d,%d)",userid,friendid);

    MySQL mysql;

    if(mysql.connect()){
        mysql.update(sql);
    }
}

vector<User> FriendModel::query(int userid){

    char sql[1024] = {0};

    sprintf(sql, "select u.id, u.name, u.state from chat_user u INNER JOIN chat_friend f ON f.friendId = u.id where f.userId = %d", userid);

    vector<User> vec;

    MySQL mysql;

    if(mysql.connect()){
        MYSQL_RES* res = mysql.query(sql);
        
        if(res != nullptr){
            MYSQL_ROW row;
            while((row = mysql_fetch_row(res)) != nullptr){
                User user;
                user.setId(atoi(row[0]));
                user.setName(row[1]);
                user.setState(row[2]);
                vec.push_back(user);
            }
            mysql_free_result(res);
        }
    }
    return vec;
}